package com.fibo.wechat;

import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.LinkedHashMap;
import java.util.Map;

/**
 * 百度api调用，采用的是sn验证
 * 如果是ip白名单验证，这个类就用不到了
 *
 * @author fibo
 * @since 2023-04-07
 */
public class CalSnUtil {

    /**
     * 计算sn跟参数对出现顺序有关，get请求请使用LinkedHashMap保存<key,value>，该方法根据key的插入顺序排序；
     * post请使用TreeMap保存<key,value>，该方法会自动将key按照字母a-z顺序排序。
     * 所以get请求可自定义参数顺序（sn参数必须在最后）发送请求，但是post请求必须按照字母a-z顺序填充body（sn参数必须在最后）。
     * 以get请求为例：<a href="http://api.map.baidu.com/geocoder/v2/?address=百度大厦&output=json&ak=yourak">...</a>，
     * paramsMap中先放入address，再放output，然后放ak，放入顺序必须跟get请求中对应参数的出现顺序保持一致。
     *
     * @param districtId 区域id
     * @param ak         百度创建应用后会有一个ak
     * @param sk         调用api采用sn验证，会有一个sk
     */
    public static String calWeatherSn(String districtId, String ak, String sk) {
        Map<String, String> paramsMap = new LinkedHashMap<>();
        // 西湖区
        paramsMap.put("district_id", districtId);
        paramsMap.put("data_type", "all");
        paramsMap.put("ak", ak);
        // 调用下面的toQueryString方法，对LinkedHashMap内所有value作utf8编码，拼接返回结果address=%E7%99%BE%E5%BA%A6%E5%A4%A7%E5%8E%A6&output=json&ak=yourak
        String paramsStr = toQueryString(paramsMap);
        // 对paramsStr前面拼接上/geocoder/v2/?，后面直接拼接yoursk得到/geocoder/v2/?address=%E7%99%BE%E5%BA%A6%E5%A4%A7%E5%8E%A6&output=json&ak=yourakyoursk
        String wholeStr = "/weather/v1/?" + paramsStr + sk;
        // 对上面wholeStr再作utf8编码
        String tempStr = URLEncoder.encode(wholeStr, StandardCharsets.UTF_8);
        return md5(tempStr);
    }

    /**
     * 对Map内所有value作utf8编码，拼接返回结果
     *
     * @param data 参数
     * @return 拼接好的参数
     */
    public static String toQueryString(Map<String, String> data) {
        StringBuilder queryString = new StringBuilder();
        for (Map.Entry<String, String> pair : data.entrySet()) {
            queryString.append(pair.getKey()).append("=");
            queryString.append(URLEncoder.encode((String) pair.getValue(), StandardCharsets.UTF_8)).append("&");
        }
        if (queryString.length() > 0) {
            queryString.deleteCharAt(queryString.length() - 1);
        }
        return queryString.toString();
    }

    /**
     * 来自stackoverflow的MD5计算方法，调用了MessageDigest库函数，并把byte数组结果转换成16进制
     *
     * @param md5 md5串
     * @return 加密
     */
    public static String md5(String md5) {
        try {
            java.security.MessageDigest md = java.security.MessageDigest.getInstance("MD5");
            byte[] array = md.digest(md5.getBytes());
            StringBuilder sb = new StringBuilder();
            for (byte b : array) {
                sb.append(Integer.toHexString((b & 0xFF) | 0x100), 1, 3);
            }
            return sb.toString();
        } catch (java.security.NoSuchAlgorithmException e) {
            e.printStackTrace();
        }
        return null;
    }

}
